FROM ANON:cuda-11.8.0-pytorch-1.13.0-torchvision-0.14.0-torchaudio-0.13.0-ubuntu-20.04
COPY . /src/nm-rnn

# shared
RUN pip3 install --upgrade pip seaborn
# specific
RUN pip3 install --upgrade gymnasium==0.29.1
RUN pip3 install --upgrade tensorboard torch-tb-profiler
# RUN pip3 install -upgrade tqdm

RUN pip3 install -e /src/dynamic-foraging-rnn